from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


def jwt_response_payload_handler(token, user=None, request=None):
    """
    自定义jwt认证成功返回数据
    """
    return {
        'token': token,
        'id': user.id,
        'username': user.username
    }


# 定义分页器
class PageNum(PageNumberPagination):
    page_size = 5  # 如果为指定每页返回的数量的情况下，默认返回五条
    page_size_query_param = "pagesize"  # 指定控制每页显示的数量
    max_page_size = 4  # 每页最多显示的数量

    # 改写分页返回方法
    def get_paginated_response(self, data):
        return Response({
            'count': self.page.paginator.count,
            'lists': data,
            'page': self.page.number,
            'pages': self.page.paginator.num_pages,
            'pagesize': self.page_size
        })


# 定义分页器
class PageNum1(PageNumberPagination):
    page_size = 5  # 如果为指定每页返回的数量的情况下，默认返回五条
    page_size_query_param = "pagesize"  # 指定控制每页显示的数量
    max_page_size = 4  # 每页最多显示的数量

    # 改写分页返回方法
    def get_paginated_response(self, data):
        return Response({
            'count': self.page.paginator.count,
            'list': data,
            'page': self.page.number,
            'pages': self.page.paginator.num_pages,
            'pagesize': self.page_size
        })


